package Intermediate_algorithm.Math;

import org.junit.Test;

import java.util.HashSet;
import java.util.Set;

/*
快乐数
编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为：
对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和。
然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。
如果这个过程 结果为1，那么这个数就是快乐数。
如果 n 是 快乐数 就返回 true ；不是，则返回 false 。

示例 1：
输入：n = 19
输出：true
解释：
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
示例 2：
输入：n = 2
输出：false
作者：LeetCode
链接：https://leetcode.cn/leetbook/read/all-about-lockup-table/fqqcs/
 */
public class _01快乐数 {

    @Test
    public void test() {
        System.out.println(isHappy(19));
    }

    public boolean isHappy(int n) {
        if(n == 1) return true;
        int temp = 0;
        for (int i = 0; i < 500; i++) {
            while (n != 0){
                temp += (n % 10) * (n % 10);
                n = n / 10;
            }
            if (temp == 1) return true;
            n = temp;
            temp = 0;
        }
        return temp == 1;
    }

    //官解：
    //ps: 链接里很好的解释了为什么不需要考虑无限增大的情况
    /*
    作者：力扣官方题解
    链接：https://leetcode.cn/problems/happy-number/solutions/224894/kuai-le-shu-by-leetcode-solution/
     */
    class Solution {
        private int getNext(int n) {
            int totalSum = 0;
            while (n > 0) {
                int d = n % 10;
                n = n / 10;
                totalSum += d * d;
            }
            return totalSum;
        }

        public boolean isHappy(int n) {
            Set<Integer> seen = new HashSet<>();
            while (n != 1 && !seen.contains(n)) {
                seen.add(n);
                n = getNext(n);
            }
            return n == 1;
        }
    }


}
